package com.atguigu.gmall.all.controller;


import com.atguigu.gmall.common.result.Result;
import com.atguigu.gmall.model.product.BaseCategoryView;
import com.atguigu.gmall.product.client.ProductFeignClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 首页
 */
@Controller
public class IndexController {

    @Autowired
    private ProductFeignClient productFeignClient;

    //首页
    @GetMapping("/")
    public String index(Model model){

        model.addAttribute("list",getData());

        return "index/index";
    }

    @Autowired
    private TemplateEngine templateEngine;

    //使用静态化程序生成首页 保存在指定的盘符下
    @GetMapping("/createHtml")
    @ResponseBody
    public Result createHtml() throws Exception{
        //参数1： 模板
        //参数2： 数据
        Context context = new Context();
        context.setVariable("list",getData());
        //参数3： 输出流    "D:\\temp\\index.html"

//        Writer writer = new OutputStreamWriter(new FileOutputStream("D:\\temp\\index.html"),
//                "utf-8");
        Writer writer = new PrintWriter("D:\\temp\\index.html","utf-8");

        templateEngine.process("index/index",context,writer);
        return Result.ok();
    }

    public List<Map> getData(){
        List<Map> listMap = new ArrayList<>();
        //所有的分类
        List<BaseCategoryView> baseCategoryViewList = productFeignClient.getBaseCategoryViewList();

        //BaseCategoryView入参  ::后面是前面的一个参数  getCategory1Id返回值
        //根据一级分类id 进行分组 返回值是一个Map
        Map<Long, List<BaseCategoryView>> baseCategoryViewListMap = baseCategoryViewList
                .stream().collect(Collectors.groupingBy(BaseCategoryView::getCategory1Id));

        //前60个数据算一个map，通过一级分类id区分
        baseCategoryViewListMap.forEach((k1,v1) -> {
            //一级分类
            Map category1Map = new HashMap();
            category1Map.put("index",k1);
            category1Map.put("categoryId",k1);
            category1Map.put("categoryName",v1.get(0).getCategory1Name());
            //二级分类
            Map<Long, List<BaseCategoryView>> baseCategoryView2ListMap = v1
                    .stream().collect(Collectors.groupingBy(BaseCategoryView::getCategory2Id));
            List<Map> listMap2 = new ArrayList<>();
            baseCategoryView2ListMap.forEach((k2,v2) -> {
                Map category2Map = new HashMap();
                category2Map.put("index",k2);
                category2Map.put("categoryId",k2);
                category2Map.put("categoryName",v2.get(0).getCategory2Name());
                //三级分类
                List<Map> listMap3 = new ArrayList<>();
                v2.forEach(baseCategoryView -> {
                    Map category3Map = new HashMap();
                    category3Map.put("categoryId",baseCategoryView.getCategory3Id());
                    category3Map.put("categoryName",baseCategoryView.getCategory3Name());
                    listMap3.add(category3Map);
                });

                category2Map.put("categoryChild",listMap3);
                listMap2.add(category2Map);
            });
            category1Map.put("categoryChild",listMap2);

            listMap.add(category1Map);
        });
        return listMap;
    }


}
